from typing import Optional
from pydantic import BaseModel


class Document(BaseModel):
    content: str
    category: str
    document_id: Optional[str] = None
